import torch
from . import _C

def create_octree(points, xyz_min, xyz_max):
    return _C.create_octree(points, xyz_min, xyz_max)

def query_grid(points, xyz_mins, xyz_maxs, sons, is_leaf_nodes):
    return _C.query_grid(points, xyz_mins, xyz_maxs, sons, is_leaf_nodes)

